package com.wanglei.ssm.aop;

import com.wanglei.ssm.domain.SysLog;
import com.wanglei.ssm.service.SysLogService;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.core.context.SecurityContext;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.userdetails.User;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.RequestMapping;

import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.util.Date;

/**
 * 系统操作日志切面类
 *
 * @author wl
 */
@Component
@Aspect
public class LogAop {

    // 请求体
    @Autowired
    private HttpServletRequest request;

    // 系统操作日志服务
    @Autowired
    private SysLogService sysLogService;

    // 访问时间
    private Date startTime;
    // 访问的类
    private Class executionClass;
    // 访问的方法
    private Method executionMethod;

    /**
     * 该切点扫描com.wanglei.ssm.controller包下所有的控制器及其方法
     */
    @Pointcut("execution(* com.wanglei.ssm.controller.*.*(..))")
    public void sysLogPoint() {
    }

    /**
     * 前置通知-主要功能是初始化访问时间，访问的类，访问的类的方法
     */
    @Before("sysLogPoint()")
    public void doBefore(JoinPoint jp) throws NoSuchMethodException, SecurityException {
        // 获取访问时间
        startTime = new Date();

        // 获取访问的类
        executionClass = jp.getTarget().getClass();

        // 获取访问的方法
        // 获取访问的方法名称
        String methodName = jp.getSignature().getName();
        // 获取访问的方法的参数
        Object[] args = jp.getArgs();
        // 无参数
        if (args == null || args.length == 0) {
            // 获取无参数方法
            executionMethod = executionClass.getMethod(methodName);
        } else {
            // 有参数，将args中的所有元素遍历，获取对应的Class,装入到一个Class[],然后获取有参数的方法
            Class[] classArgs = new Class[args.length];
            for (int i = 0; i < args.length; i++) {
                classArgs[i] = args[i].getClass();
            }
            executionMethod = executionClass.getMethod(methodName, classArgs);
        }
    }

    /**
     * 主要获取日志中的其它信息，执行时长、ip、url
     */
    @After("sysLogPoint()")
    public void doAfter() throws Exception {
        // 获取类上的@RequestMapping对象
        RequestMapping classRequestMappingAnnotation = (RequestMapping) executionClass.getAnnotation(RequestMapping.class);
        if (classRequestMappingAnnotation != null) {
            // 获取方法上的RequestMapping对象
            RequestMapping methodRequestMappingAnnotation = executionMethod.getAnnotation(RequestMapping.class);
            if (methodRequestMappingAnnotation != null) {
                // 获取rul,他的值是类上的@RequestMapping的vaLue+方法上的@RequestMapping的vaLue
                String url = classRequestMappingAnnotation.value()[0] + methodRequestMappingAnnotation.value()[0];
                // 获取ip
                String ip = request.getRemoteAddr();
                // 获取用户名，通过securityContext获取，securityContext也可以从request.getSession中获取
                SecurityContext context = SecurityContextHolder.getContext();
                // request.getSession().getAttribute("SPRING_SECURITY_CONTEXT");
                String username = ((User) (context.getAuthentication().getPrincipal())).getUsername();
                // 获取访问时长
                long executionTime = new Date().getTime() - startTime.getTime();

                // 创建系统操作日志对象
                SysLog sysLog = new SysLog();

                // 封装sysLog
                sysLog.setExecutionTime(executionTime);
                sysLog.setUrl(url);
                sysLog.setIp(ip);
                sysLog.setUsername(username);
                sysLog.setVisitTime(startTime);
                sysLog.setMethod("[类名]" + executionClass.getName() + "[方法名]" + executionMethod.getName());
                // 存储sysLog到数据库中
                sysLogService.save(sysLog);
            }
        }
    }

}
